# coding=utf-8
# Copyright 2024 The Google Research Authors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

r"""Ground truth values for `german_credit_numeric_with_test_logistic_regression`."""

import numpy as np

PARAMS_MEAN: np.ndarray = np.array([
    -0.7840015955133335,
    0.400315037737037,
    -0.51759129461422,
    0.15862982613366433,
    -0.3811638646018738,
    -0.21621973473123224,
    -0.06243553779702192,
    -0.030273842530384378,
    0.23730499623414217,
    -0.092318948744981,
    -0.24396488902096586,
    0.200773499007345,
    0.001548132172336506,
    -0.24511387684026403,
    -0.363895442939817,
    0.257972537109941,
    -0.3246438485379358,
    0.30818215617328587,
    0.3519937023774831,
    0.016560249649862527,
    -0.144083660572314,
    -0.0681721252940579,
    0.022626810812656016,
    0.022305200367755268,
    -1.250468260058,
]).reshape((25,))

PARAMS_MEAN_STANDARD_ERROR: np.ndarray = np.array([
    7.145948954378882e-05,
    8.926044293708226e-05,
    7.864849510736585e-05,
    9.337530361982083e-05,
    7.436576683637909e-05,
    7.325530641330479e-05,
    6.252305699291936e-05,
    7.361196397749715e-05,
    8.923233426264415e-05,
    7.961651804356138e-05,
    6.115929317903135e-05,
    7.812106675421792e-05,
    6.591071264658985e-05,
    7.773315270434295e-05,
    0.00010057705392900466,
    6.51790615809859e-05,
    8.228590503808683e-05,
    0.0001041122548892971,
    9.516491972388064e-05,
    0.00013333131276103226,
    0.00013533637619651053,
    7.72519718086808e-05,
    0.0001141409468837317,
    0.00010998107330347721,
    7.803022288185956e-05,
]).reshape((25,))

PARAMS_STANDARD_DEVIATION: np.ndarray = np.array([
    0.10548603830182482,
    0.12451140629817728,
    0.1139308312460883,
    0.1282248231221989,
    0.11211430742087578,
    0.10841929494488367,
    0.09649310060117368,
    0.11003344753019306,
    0.12287935516209522,
    0.11448296328010718,
    0.0932888909057452,
    0.11260554280235917,
    0.1006008735122863,
    0.11243236937089811,
    0.14308267503943892,
    0.09861216388674257,
    0.12234647557031404,
    0.13909620678939455,
    0.12807242910866112,
    0.1652531608513733,
    0.16851431236503592,
    0.11074481976820545,
    0.14979713194200223,
    0.14432957403434168,
    0.10985042225840513,
]).reshape((25,))

PER_EXAMPLE_TEST_NLL_MEAN: np.ndarray = np.array([
    0.6676377528830668,
    1.0563918366646665,
    0.06147464278592,
    1.21124353468,
    2.2240296006273335,
    0.27117351853253335,
    0.1323875656194,
    0.03285735328183333,
    0.49769438302326663,
    0.4336187880435333,
    0.9005376817706667,
    0.13946746141606664,
    0.6259795288860001,
    0.03642125307446667,
    0.09413545132973333,
    0.19902101347920004,
    0.019540203404965865,
    0.9116032830735332,
    0.31531664134686666,
    0.14176906819506668,
    0.8421425040193334,
    0.09510488055960001,
    1.1791076526044,
    1.4346412537713333,
    0.5937367521233333,
    0.07496533603097333,
    1.1131550990646666,
    0.4565358542073333,
    0.2415765855382666,
    0.5904098587779335,
    1.4085609455266666,
    0.2670111454274667,
    0.09777276934693334,
    0.24906320014246663,
    0.044339795840946675,
    0.16471982069899332,
    0.9228005999800001,
    1.0761299071133332,
    2.002314929976,
    0.18302993332586664,
    0.03343035386853334,
    1.5497237726958002,
    0.1505979366854,
    2.4464505293279997,
    0.9430558560438665,
    0.19380590296833336,
    0.14359789863897338,
    0.28468585157233334,
    0.2502991843533333,
    0.13948084455410667,
    0.29862440739826673,
    0.08283974787221533,
    0.033538798546780005,
    0.036352743764526664,
    2.35025243533,
    0.30299640837913333,
    0.15610586328953333,
    0.4016297882810667,
    0.2136250337526667,
    0.012479131087308802,
    0.21164421223812666,
    0.05871792201371999,
    1.1755206884986666,
    1.5072441609133331,
    0.13510318924270667,
    0.7023036754367998,
    0.1438474058296,
    0.1616616992896,
    0.08644523323673334,
    0.017891975059682666,
    0.6694851552053332,
    0.30804364622046665,
    0.17464109108818,
    0.2988630021202,
    0.5182297508722666,
    1.3170283097986666,
    0.8040132058333332,
    0.15544822176673334,
    0.15157077882093334,
    0.08594285810820002,
    0.4500119102890666,
    0.014249504025653333,
    0.3934893661742,
    0.7739916114733333,
    0.7876895747913334,
    0.02888342652540667,
    0.5169861575864667,
    0.2427756960006,
    2.0731907922760002,
    0.684511006822,
    1.062085960204,
    1.7900518993186663,
    1.8456633062959997,
    0.7590121005086667,
    1.8167627313353336,
    0.7332618647526666,
    0.6676579592956666,
    0.11618304352013334,
    0.40235771636479994,
    0.28280329096375334,
    0.3123350933907333,
    0.0466781022153,
    0.844659505502,
    0.18405363776353328,
    0.27885551452440005,
    0.10013553250457999,
    1.0669073142866665,
    0.44855563991826664,
    0.06465661679993333,
    1.4827066064493333,
    0.029121625610460007,
    0.23587506272308664,
    0.45374029009100003,
    0.018437547708468664,
    0.39981253825239993,
    0.427037295351,
    0.5685083724378666,
    0.03912716225389333,
    1.9805831194493329,
    0.0870497426362,
    2.0397331935944663,
    0.011064119167389332,
    0.514809726026,
    0.045082487163146674,
    0.7828713731004001,
    0.07935265807644667,
    0.34912527879413335,
    0.26655120297973334,
    1.853849987518667,
    0.1697082297292,
    0.08525303634364667,
    0.48267901876793334,
    1.5624223892293334,
    1.2159411148514,
    0.520824709119,
    0.5569574654365332,
    0.09398086563493334,
    0.5028593683082666,
    0.2557006302807333,
    0.8420803272599999,
    1.896675378878666,
    0.023562558148793333,
    0.2581881365926667,
    0.23244739929453334,
    0.16398151215173334,
    0.33947239131086665,
    0.10633387724557467,
    0.02141208295166,
    0.04894393964942667,
    1.08348128667,
    0.1284519956813333,
    0.012139784516430666,
    1.5331728439373333,
    0.17386678180853335,
    0.6972674417229333,
    0.10317225512619999,
    0.07647104436221333,
    0.16015971810344667,
    0.22036951102006666,
    0.8250919672566,
    0.2075223930624,
    0.2813082428495334,
    0.6649578006172,
    0.062394601926106665,
    0.10583004955880002,
    0.11500913738891998,
    0.3706603048588667,
    0.2433343802969333,
    2.8674432726766668,
    1.2040085889599332,
    3.61667726062,
    0.0811144073616,
    0.674240429556,
    0.27367775499506675,
    0.09359528860066665,
    0.18137552701246668,
    1.5910488913526666,
    0.11767520494258668,
    0.5819564133268667,
    0.4026187853524667,
    0.6920757970933333,
    0.024370633428913338,
    0.5677278576139999,
    0.06377713480160667,
    0.6371189775086666,
    2.6905847513766665,
    1.2818540711193334,
    0.11514429121986665,
    2.179863633118667,
    0.08239728789806666,
    0.2637455811814,
    0.12546935465113335,
    0.9836544749286669,
    0.5340772018713998,
    1.0618829274519999,
    0.06767648196639334,
    0.7784622093609334,
    0.09143889027398,
    0.05407337528319334,
    0.05227821082556666,
    0.17489411137606667,
    1.1078904800917335,
    0.23244104681386663,
    0.5832169102984667,
    0.017870858438097933,
    0.6140891374262667,
    0.31701403076219997,
    0.7224888543819998,
    0.08030680525068666,
    0.17129896498966665,
    0.19497950240133335,
    2.237901076784,
    0.4629127709923999,
    2.56617577,
    1.5859860328686666,
    0.05376730785467334,
    0.8651224069793333,
    0.46897436388406666,
    0.18120385610786666,
    0.16410784306953335,
    0.35217479708646665,
    0.15814178083933333,
    0.7411339456953333,
    0.22506441338960004,
    0.8245026021327334,
    0.17958751562286665,
    0.16019508545046665,
    0.179413884027348,
    0.5378815095477867,
    0.3004532209232667,
    0.18157069317479996,
    0.4797436064586,
    0.07048604039988,
    0.056063725959,
    0.6341535312248,
    1.2950755752771332,
    0.1596867688008,
    1.266411790942,
    0.49756629725386664,
    0.3363811214313333,
    0.19471923362760005,
    0.16558331236946663,
    0.09192059330037998,
    0.17067540127846664,
    0.03566094747842666,
    0.2364851875470666,
    0.1393543449372,
    0.02136023027062,
    1.5424973951,
    0.01433884564011747,
]).reshape((250,))

PER_EXAMPLE_TEST_NLL_MEAN_STANDARD_ERROR: np.ndarray = np.array([
    0.0001739940608812719,
    0.00020163524983706155,
    2.1292173457210853e-05,
    0.00018609714528661502,
    0.0003217666644210296,
    7.723003777067757e-05,
    2.940831858535854e-05,
    1.0757613189951844e-05,
    0.00015712710880501157,
    9.112531744012974e-05,
    0.00017422844041749386,
    4.2106202830298475e-05,
    0.0001293634327565147,
    1.3864649432787134e-05,
    2.480892256993764e-05,
    5.3364516702677314e-05,
    1.3114153622043694e-05,
    0.00023119609628024238,
    9.374242780820216e-05,
    2.483371935759963e-05,
    0.0001465682443745883,
    2.3058996666333687e-05,
    0.00028611642435157254,
    0.00020108914378519227,
    0.00012075429410997538,
    2.197965761546146e-05,
    0.0002346894198246816,
    7.863776943984514e-05,
    5.7312234376489966e-05,
    0.00013811901308179866,
    0.0002235574746000571,
    6.339011466764027e-05,
    2.8432419828845013e-05,
    8.623522571134599e-05,
    1.5929387829825274e-05,
    7.861040981817386e-05,
    0.00018990449109748653,
    0.00018900030495004762,
    0.00022939338929135703,
    6.104549341775209e-05,
    1.5165934121356982e-05,
    0.0005326030902263552,
    5.582611035579959e-05,
    0.000444999113889993,
    0.0002451222806364263,
    6.09001183061621e-05,
    5.35019179470894e-05,
    4.776892274137994e-05,
    7.879533029550534e-05,
    5.405366300062715e-05,
    0.0001260811555935981,
    5.101850526542662e-05,
    1.3731859671221722e-05,
    1.0767637953721322e-05,
    0.0003234901376739195,
    8.756447146437442e-05,
    4.3788737606210494e-05,
    8.421655466148214e-05,
    5.673543644994368e-05,
    8.693418900628884e-06,
    8.137584351154879e-05,
    2.4396811457622286e-05,
    0.00015444202733949508,
    0.00019063058962883164,
    5.228079463717763e-05,
    0.00016884365525170157,
    4.372138203286276e-05,
    4.417624859612022e-05,
    2.4098382350084955e-05,
    8.033736576511179e-06,
    0.00010469514925047667,
    0.00012162889589512976,
    7.314145500348335e-05,
    6.88537058351773e-05,
    0.0001367973903079786,
    0.0001389173421591665,
    9.944129616230373e-05,
    4.2187275488978155e-05,
    3.709794133567816e-05,
    3.01935590755372e-05,
    0.00010121296599475559,
    6.329903174657492e-06,
    8.148331329296174e-05,
    0.00012744624005942722,
    0.00014167020834783436,
    1.1285077390404204e-05,
    0.0001315587085505272,
    6.259085872737498e-05,
    0.0002714933070077859,
    0.0001186766559887053,
    0.00016671959002417032,
    0.00030161560627139705,
    0.0002478509422509563,
    0.0001965085462464409,
    0.00027266358363204425,
    0.00013845091126913101,
    0.00014849847486553882,
    2.375883749008041e-05,
    9.340905833984335e-05,
    0.00011425698270333898,
    8.143011597423663e-05,
    1.5066642844908848e-05,
    0.0001755237586626034,
    4.513688361807575e-05,
    6.806476966866476e-05,
    3.7164809235062734e-05,
    0.0002445252705520134,
    0.00017670870015122015,
    1.5404943883936482e-05,
    0.00030464178749218976,
    1.0522933691987155e-05,
    0.0001043567822908624,
    8.432425146657573e-05,
    8.557525256760597e-06,
    0.00012582390826741305,
    0.0001235789025702085,
    0.00016906263869989645,
    1.5717286058180753e-05,
    0.0003115574243751742,
    2.2913201030281247e-05,
    0.000539241966157952,
    5.074720612028874e-06,
    0.00011839472612401204,
    1.6064913202285635e-05,
    0.0002901412526699559,
    2.5075107499449934e-05,
    0.00012586513914087042,
    5.530925398325046e-05,
    0.00022030083660134057,
    6.281987141895195e-05,
    3.0148972060868003e-05,
    0.00016836719504414272,
    0.00027903882181692246,
    0.00031977359624675025,
    0.00010657251858439663,
    0.00012069187505313911,
    2.2229389636130503e-05,
    0.00011527033457482101,
    6.713122274086082e-05,
    0.00017347300573147345,
    0.00027722564502119003,
    8.538313840808973e-06,
    5.2221499040865144e-05,
    6.1295269553753e-05,
    4.008203335570577e-05,
    6.38766735515828e-05,
    6.424354922467589e-05,
    7.405188360628005e-06,
    1.9239810016551338e-05,
    0.00013439510747763865,
    2.6133694383828613e-05,
    6.287415954867073e-06,
    0.00024589896759013034,
    3.609618109285227e-05,
    0.0002106807316391339,
    3.8630382352958825e-05,
    4.360747520585121e-05,
    6.084474504523915e-05,
    5.54790470261506e-05,
    0.00019475802636188786,
    7.961484203721212e-05,
    8.235697076719796e-05,
    0.0003018692799292702,
    1.9972578100442782e-05,
    4.2428680143952946e-05,
    3.629770454396898e-05,
    9.197568939920976e-05,
    5.1614902233405924e-05,
    0.0005094646646798805,
    0.00032047834573919266,
    0.00037405709396582217,
    1.6827064161231046e-05,
    0.0001007263900177916,
    8.777764111937887e-05,
    2.1223737960095714e-05,
    6.235795446886042e-05,
    0.0003795182088485498,
    5.316669337252334e-05,
    0.0001881828624414501,
    0.00011766592744909179,
    0.0001364535510595205,
    6.831454879054338e-06,
    0.0001046766213810089,
    2.824908023400625e-05,
    0.00013639122306556225,
    0.0004579075085077069,
    0.00024267899685428518,
    2.6696247204178917e-05,
    0.00028641578188116185,
    2.181013622375215e-05,
    7.50362572120243e-05,
    3.864387010722998e-05,
    0.00017896636047431502,
    0.00012446395217351736,
    0.00015600558931697225,
    1.7988951980283884e-05,
    0.0001856740789048076,
    4.006225388522752e-05,
    1.5307048903929884e-05,
    2.2182137439318195e-05,
    6.452075408721819e-05,
    0.0003551713303251003,
    5.5274449871497076e-05,
    0.00014715665858097208,
    1.425876239579092e-05,
    0.00014440247723581764,
    0.00010842137901495889,
    0.00012270853500305763,
    3.1943121344853974e-05,
    5.063248510892966e-05,
    5.689358237662882e-05,
    0.00026004110729502445,
    0.00010938496871653427,
    0.0002863129787207158,
    0.00022159220335728961,
    1.4716664823617717e-05,
    0.00016732334440650886,
    0.0001279428723532837,
    6.379901608269634e-05,
    4.699839696039594e-05,
    9.518327572379441e-05,
    3.921911078861345e-05,
    0.00014170905305376294,
    6.055091723430014e-05,
    0.00019920305529803976,
    5.044742747316595e-05,
    4.817793181127815e-05,
    9.688986991682235e-05,
    0.0002444721541455388,
    0.00012555824984206515,
    4.5294957687018484e-05,
    0.00016297526551477896,
    2.022401493080536e-05,
    2.1083907274099258e-05,
    0.00017227929118952988,
    0.00046505244844665034,
    4.8251539797019965e-05,
    0.00018345728336497533,
    0.0001562878946012209,
    7.777253658474783e-05,
    5.022119838918065e-05,
    4.771904132185369e-05,
    3.9038792226277236e-05,
    3.621722008814638e-05,
    1.6661474015598225e-05,
    5.386668362814161e-05,
    3.588350396993548e-05,
    8.12253449002022e-06,
    0.00024483538667042067,
    9.87082083450872e-06,
]).reshape((250,))

PER_EXAMPLE_TEST_NLL_STANDARD_DEVIATION: np.ndarray = np.array([
    0.2522181029185521,
    0.2834641220089314,
    0.02926107974140917,
    0.26030629081882095,
    0.4543599331181657,
    0.10729980088307164,
    0.04314326745216322,
    0.015055248675176103,
    0.22202686506682334,
    0.13265281508702173,
    0.259199156179449,
    0.05902563127941163,
    0.18484552114273947,
    0.019045214663878436,
    0.03472025993647343,
    0.07517745602885403,
    0.01743870496000488,
    0.31970847675015535,
    0.13513334761687165,
    0.0360162387995767,
    0.2152420705928944,
    0.03261283037306118,
    0.40328823400012836,
    0.3029075188901285,
    0.17634443146483442,
    0.03079853432667065,
    0.3347292615301008,
    0.11363993538129931,
    0.0829744516293029,
    0.19178677139822015,
    0.3138040877551249,
    0.09103273319664931,
    0.03990680307458304,
    0.11621034785691509,
    0.02168938430286234,
    0.10494564398841148,
    0.2697782798276046,
    0.2809981765352381,
    0.33826025333417087,
    0.08527464090598215,
    0.02013869140299509,
    0.7340643096461826,
    0.0755760632468376,
    0.6166341024815631,
    0.33565326427804376,
    0.08326989288167355,
    0.07298192591158298,
    0.06956033666060876,
    0.1134299739241665,
    0.07363161774294853,
    0.16252538534994082,
    0.06860657963553253,
    0.01826003605773262,
    0.014960244003309678,
    0.4594147583395688,
    0.11935102279778056,
    0.06152333923382246,
    0.12184224114950619,
    0.0788622021760295,
    0.011261755486630755,
    0.1104394229477395,
    0.0332456458330892,
    0.22328177422585566,
    0.279718405272822,
    0.07186393506000024,
    0.24199131574533844,
    0.06110268849361927,
    0.061014513146705016,
    0.03308301939452921,
    0.01077342429575472,
    0.147846730383672,
    0.1685245584860415,
    0.0981665770430238,
    0.09957936623306526,
    0.1959710000322918,
    0.19725381567718536,
    0.14261421978271285,
    0.05907703965512714,
    0.05136680563388786,
    0.03981222154652232,
    0.14718421824483727,
    0.008391863584393726,
    0.11979128016470861,
    0.1855093447891408,
    0.20554542053053143,
    0.015240579201713251,
    0.18742829252618312,
    0.08782414869282372,
    0.3982397760118861,
    0.16837935586361613,
    0.24435096884328336,
    0.43247763844343207,
    0.3484196160715191,
    0.2562102851723328,
    0.3914698896899686,
    0.2049327243967869,
    0.21105364673730845,
    0.03471664634338528,
    0.13152803857144746,
    0.15620013709156227,
    0.11305384196085552,
    0.021255222903239362,
    0.2557899410132597,
    0.06224835663970883,
    0.0958189768668265,
    0.05092886849812034,
    0.3414454029307741,
    0.23165550560004497,
    0.021523182824606402,
    0.4408250199971687,
    0.014236322080977101,
    0.14091660979139325,
    0.1237661281303577,
    0.011351253023444467,
    0.1702473615209758,
    0.1705313122628505,
    0.23613399486780856,
    0.021526740802224976,
    0.4416837108487751,
    0.03256909633028463,
    0.7654962961428834,
    0.006808286511684286,
    0.17017341513533643,
    0.022382510598642408,
    0.4144773760740493,
    0.035008887482063294,
    0.173589623481678,
    0.07982139366977932,
    0.3267223179675998,
    0.08555152437807012,
    0.04012356774858454,
    0.23291240012876088,
    0.39695638918397924,
    0.4457695632470829,
    0.1539555273931847,
    0.17353088213091666,
    0.03143700864708454,
    0.1642021675954773,
    0.09527101370619327,
    0.2514740189354367,
    0.39027642958688363,
    0.011719462828348131,
    0.07518976714619477,
    0.08448002916165034,
    0.056738922938064595,
    0.08741807923080838,
    0.08620667623694638,
    0.01005296300088056,
    0.026091589771737166,
    0.19767313297721173,
    0.03838587088479336,
    0.008193866275652995,
    0.3589657570433794,
    0.05221422981390146,
    0.2919360397657842,
    0.052502170696070646,
    0.05838738461697508,
    0.08208751393864547,
    0.07823645182806639,
    0.2819593312911328,
    0.10757957191800056,
    0.11570176721490195,
    0.4135025429917647,
    0.0278303643035333,
    0.05731020552991488,
    0.05035574702492408,
    0.12827378752528248,
    0.07218628167778733,
    0.7254650839132835,
    0.4555678769117736,
    0.5423091363871974,
    0.024124517165294034,
    0.13979415214165458,
    0.11884789988880406,
    0.029883416219311416,
    0.08519480157044465,
    0.5226045020650529,
    0.07153609529022734,
    0.2613783529987607,
    0.16721839928567545,
    0.1976190596062358,
    0.009574714199986362,
    0.1470218437614422,
    0.03822346774348527,
    0.18788758906438932,
    0.6261795513504964,
    0.33789474381820905,
    0.038479258249580366,
    0.4189182870493637,
    0.029892351618817016,
    0.10350714061006583,
    0.05182024941832562,
    0.2609367454759406,
    0.17962732899291417,
    0.23102244363642868,
    0.025355171528914717,
    0.27126842211518054,
    0.054017258148423286,
    0.02154906045210906,
    0.030201360312077586,
    0.08559650600193522,
    0.5235079635752238,
    0.0773316020364042,
    0.19969936781698,
    0.018242861262276273,
    0.20570033977003233,
    0.14569432112403552,
    0.17842774158939914,
    0.04258403268024073,
    0.07000211209956896,
    0.0801650874956575,
    0.38059559779556906,
    0.15909755531492592,
    0.4155656215984858,
    0.30931489815917995,
    0.020768978377366583,
    0.23660370725962093,
    0.18131535853382996,
    0.08362721617420724,
    0.06620600305308413,
    0.1344241059690699,
    0.05574788529381454,
    0.20103502867544223,
    0.08454997321615257,
    0.2888456932255811,
    0.07150774543075353,
    0.0678990590700574,
    0.1351880532051895,
    0.3468735395356388,
    0.17278963262434827,
    0.062400922014403705,
    0.22453141840780258,
    0.02798075840874682,
    0.02903950127632879,
    0.24563616216033513,
    0.6712662492325577,
    0.06689011192740761,
    0.2661760712790212,
    0.2175491554510261,
    0.11077352888700513,
    0.06971056784949266,
    0.06789404596820904,
    0.051968047776602265,
    0.050597623611448574,
    0.02176617566056692,
    0.07451066659161683,
    0.05015625209218828,
    0.011320366358743533,
    0.3613127584022934,
    0.012751681166812337,
]).reshape((250,))

TEST_NLL_MEAN: np.ndarray = np.array([
    136.69764967666666,
]).reshape(())

TEST_NLL_MEAN_STANDARD_ERROR: np.ndarray = np.array([
    0.0030785994468437807,
]).reshape(())

TEST_NLL_STANDARD_DEVIATION: np.ndarray = np.array([
    3.8472656957782783,
]).reshape(())
